# 引入unittest模块
import unittest

# 由于测试函数在UserController.py文件中，我们需要导入该模块
# 假设UserController.py在同一个目录下，可以这样导入
from UserController import my_func

# 定义测试类，继承自unittest.TestCase
class TestMyFunc(unittest.TestCase):

    # 测试my_func函数
    def test_my_func(self):
        # 调用my_func函数
        # result = my_func()
        # 断言结果为None
        self.assertEqual(2, my_func(1))
        self.assertEqual(3, my_func(-1))

        for i in range(-100, 100):
            if i == 1 or i == -1:
                continue
            self.assertEqual(1, my_func(i))

# 这个if语句是单元测试模块的标准写法，用于保证当该脚本被直接运行时，单元测试将自动执行
if __name__ == '__main__':
    unittest.main()
